#include "mystdio.h"


#include <unistd.h>
#include <string.h>
#include <fcntl.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <string.h>
#include <stdlib.h>

_FILE* _fopen(const char *filename,char *flag)
{

    int fd = 0;
    if(strcmp(flag , "w") == 0)
    {
        
        fd = open(filename , O_CREAT|O_WRONLY|O_TRUNC,0666);
    }
   else if(strcmp(flag , "r") == 0)
   {
        fd = open(filename, O_RDWR);
   }
   else if(strcmp(flag , "a") == 0)
   {
       fd = open(filename,O_CREAT|O_APPEND);
   }
    else {
        return NULL;
    }
    


    if(fd == -1)
    {
        return NULL;
    }

    _FILE* fp = (_FILE*)malloc(sizeof(_FILE));
    fp->fileno = fd;
    fp->flag = FLUSH_LINE;
    fp->out_pos = 0;

    return fp;
} 

int  _fwrite(_FILE* fp,const char* s,int len)
{
    memcpy(&fp->outbuffer[fp->out_pos],s,len);
    fp->out_pos += len;


    if(fp->flag & FLUSH_NOW)
    {
        write(fp->fileno,s,len);   
    }
    else if(fp->flag & FLUSH_LINE)
    {
        if(fp->outbuffer[fp->out_pos-1] == '\n')
        {
            write(fp->fileno,s,len);
        }
    }
    else if(fp->flag& FLUSH_ALL) {
        write(fp->fileno,fp->outbuffer,fp->out_pos);

    }

    return len;
}

void _fflush(_FILE* fp)
{
    if(fp->out_pos > 0)
    {
        write(fp->fileno,fp->outbuffer,fp->out_pos);
        fp->out_pos = 0;
    }
    
}

void _close(_FILE* fp)
{
    if(fp == NULL) return ;
    _fflush(fp);
    close(fp->fileno);
    free(fp);
}


